package com.zf.nospring.rabbitmq.delay.ttl;

import com.rabbitmq.client.*;
import com.zf.nospring.rabbitmq.utils.RabbitUtils;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.concurrent.TimeoutException;

/**
 * 延迟消息实现方式有两种：
 * 1、通过 TTL + 死信方式 实现延迟消息
 * 2、通过 https://github.com/rabbitmq/rabbitmq-delayed-message-exchange 插件
 * <p>
 * 这里使用 TTL + 死信方式：
 * 将消息带上TTL投递到正常队列中（或者给正常队列添加TTL），同时给正常队列配置死信队列，当消息达到TTL时间成为死信后会被投递的死信队列中，我们消费死信队列就实现了消息的延迟消费
 * <p>
 * 缺陷：
 * 由于rabbitmq自身的延迟消息 只会检查队列的队首消息是否过期；
 * 如果其他消息已经过期了，但是队首消息还没有过期，那么rabbitmq并不会把其他的过期的消息移除队列，过期消息依然被保留在队列中，
 * 直到过期的消息成为队首，rabbitmq才会将过期消息移出队列，将之投入死信队列或者丢弃，然后继续检查队首是否过期。
 * <p>
 * 所以对于同一个队列，此种方式只能实现统一时间的延迟消息，不能个性化每个消息的延迟时间。
 */
public class DelayMessageProducerByTtl {
    public static void main(String[] args) throws IOException, TimeoutException {
        Connection connection = RabbitUtils.getDefaultConnection();
        Channel channel = connection.createChannel();

        // 声明一个交换机作为 死信交换机，这里使用 Direct交换机
        String testDelayDeadExchange = "test_delay_dead_exchange";
        String testDelayDeadRouteKey = "test_delay_dead_route";
        channel.exchangeDeclare(testDelayDeadExchange, BuiltinExchangeType.DIRECT, true, false, null);

        //声明一个队列作为 死信队列，用来接收过期的消息
        String testDelayDeadQueue = "test_delay_dead_queue";
        channel.queueDeclare(testDelayDeadQueue, true, false, false, null);

        // 绑定死信队列和死信交换机
        channel.queueBind(testDelayDeadQueue, testDelayDeadExchange, testDelayDeadRouteKey);

        // 声明正常队列，并配置 死信交换机 和 死信的路由键
        String testDelayNormalQueue = "test_delay_normal_queue";
        HashMap<String, Object> queueArgs = new HashMap<>();
        queueArgs.put("x-dead-letter-exchange", testDelayDeadExchange);
        queueArgs.put("x-dead-letter-routing-key", testDelayDeadRouteKey);
        // 给整给队列设置 过期时间，消息入队后将在 20s 后过期，"x-message-ttl" 参数的单位是毫秒
        queueArgs.put("x-message-ttl", 20000);
        channel.queueDeclare(testDelayNormalQueue, true, false, false, queueArgs);

        //发送消息，使用默认交换机向正常队列发送消息
        for (int i = 1; i <= 10; i++) {
            // 指定 消息的过期时间是 5s，队列和消息同时指定TTL时，以小的时间为准
            AMQP.BasicProperties properties = new AMQP.BasicProperties.Builder()
                    .expiration("5000")
                    .build();
            channel.basicPublish("", testDelayNormalQueue, properties, ("你啊" + i).getBytes(StandardCharsets.UTF_8));
        }

        channel.close();
        connection.close();
    }
}
